package com.wqz.dynamic;

public class KnapsackProblem {
    public static void main(String[] args) {
        //物品重量
        int[] w = {1, 4, 3};
        //物品价值
        int[] val = {1500, 3000, 2000};
        //背包容量
        int m = 4;
        //物品个数
        int n = val.length;

        int[][] v = new int[n + 1][m + 1];
        //path记录的是放入背包的物品的情况，新放入背包的物品为1，表示放入背包
        //背包里的情况不变，取上一个策略的结果的话，为0
        int[][] path = new int[n + 1][m + 1];

        //初始化第一行和第一列
        for (int i = 0; i < v.length; i++) {
            //第一列为零
            v[i][0] = 0;
        }
        for (int i = 0; i < v[0].length; i++) {
            //第一行为零
            v[0][i] = 0;
        }

        /*i 和 j 从1开始是因为不用处理背包重量为0，和放入背包的物品数量为0的情况，都是为零的*/
        for (int i = 1; i < v.length; i++) {
            for (int j = 1; j < v[0].length; j++) {
                if (w[i - 1] > j) {
                    //物品的重量大于背包的重量时,上一个方案的最优解直接写入
                    v[i][j] = v[i - 1][j];
                } else {
                    //物品的重量小于等于背包的重量时,
                    /*v[i][j] = Math.max(v[i - 1][j], val[i - 1] + v[i - 1][j - w[i - 1]]);*/

                    //path记录的是放入背包的物品的情况，新放入背包的物品为1，表示放入背包
                    if (v[i - 1][j] < val[i - 1] + v[i - 1][j - w[i - 1]]) {
                        v[i][j] = val[i - 1] + v[i - 1][j - w[i - 1]];
                        path[i][j] = 1;
                    } else {
                        //背包里的情况不变，取上一个策略的结果的话
                        v[i][j] = v[i - 1][j];
                    }
                }
            }
        }

        for (int i = 0; i < v.length; i++) {
            for (int j = 0; j < v[i].length; j++) {
                System.out.print(v[i][j] + " ");
            }
            System.out.println();
        }

        int i = path.length - 1;
        int j = path[0].length - 1;
        while (i > 0 && j > 0) {
            if (path[i][j] == 1) {
                System.out.printf("第%d个商品放入到背包\n", i);
                j -= w[i - 1];
            }
            i--;
        }
    }
}
